grouped_shapes = [
    (1, 1024, 14, 14, 512, 1, 1, 1, 0, 1, 4),
    (1, 1024, 14, 14, 512, 1, 1, 1, 0, 1, 8),
    (1, 1024, 14, 14, 512, 1, 1, 1, 0, 1, 16),
    (1, 1024, 14, 14, 512, 1, 1, 1, 0, 1, 32),
]

grouped_shapes_ = [
    (1, 64, 112, 112, 192, 3, 3, 1, 1, 1, 16),  # conv2   1
    (1, 192, 56, 56, 128, 1, 1, 1, 0, 1, 16),  # conv3   2
    (1, 128, 56, 56, 256, 3, 3, 1, 1, 1, 16),  # conv4   3
    (1, 256, 56, 56, 256, 1, 1, 1, 0, 1, 16),  # conv5   4
    (1, 256, 56, 56, 512, 3, 3, 1, 1, 1, 16),  # conv6   5
    (1, 512, 28, 28, 256, 1, 1, 1, 0, 1, 16),  # conv7   6
    (1, 256, 28, 28, 512, 3, 3, 1, 1, 1, 16),  # conv8   7

    (1, 512, 28, 28, 512, 1, 1, 1, 0, 1, 16),  # conv15      8
    (1, 512, 28, 28, 1024, 3, 3, 1, 1, 1, 16),  # conv16     9
    (1, 1024, 14, 14, 512, 1, 1, 1, 0, 1, 16),  # conv17    10
    (1, 512, 14, 14, 1024, 3, 3, 1, 1, 1, 16),  # conv18     11

    (1, 1024, 14, 14, 1024, 3, 3, 1, 1, 1, 16),  # conv21   12
    (1, 1024, 14, 14, 1024, 3, 3, 2, 1, 1, 16),  # conv22   13
    (1, 1024, 7, 7, 1024, 3, 3, 1, 1, 1, 16),  # conv23     14
]